package Einheiten;

import Einheiten.Attribute.EditorAttribut;
import Einheiten.Attribute.Text;
import Schnittstellen.RTSObject;
import java.util.HashMap;

/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */
/**
 *
 * @author Geez
 */
public class Einheit extends RTSObject {

    private HashMap<Integer, EditorAttribut> attribute;

    public Einheit() {
        attribute = new HashMap<Integer, EditorAttribut>();
    }

    public void addAttribut(EditorAttribut editorAttribut) {
        this.attribute.put(editorAttribut.getAttributTyp(), editorAttribut);
    }

    public void removeAttribut(int attributsTyp) {
        this.attribute.remove(attributsTyp);
    }

    public void clearAttribute() {
        attribute.clear();
    }

    public boolean hasAttributTyp(int attributstyp) {
        return attribute.containsKey(attributstyp);
    }

    public EditorAttribut getAttribut(int attributstyp) {
        return this.attribute.get(attributstyp);
    }
}
